from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
import json

from constants.callback_events import EXAM_TYPE_EGE, EXAM_TYPE_OGE, SELECT_EXAM_ACTION


def create_exam_type_selection_keyboard() -> InlineKeyboardMarkup:
    keyboard = InlineKeyboardMarkup(row_width=2)

    keyboard.add(
        InlineKeyboardButton(
            text="ЕГЭ",
            # TODO: make a custom encoder
            callback_data=f"{SELECT_EXAM_ACTION}:{EXAM_TYPE_EGE}"
        )
    )

    keyboard.add(
        InlineKeyboardButton(
            text="ОГЭ",
            # TODO: make a custom encoder
            callback_data=f"{SELECT_EXAM_ACTION}:{EXAM_TYPE_EGE}"
        )
    )

    return keyboard
